package yuangong.dao;

import java.util.ArrayList;
import java.util.HashMap;

public class BaseIndexDao {
    private final HashMap<String, ArrayList<String>> index = new HashMap<>();

    public void add(String id, String key) {
        if (index.containsKey(key)) {
            index.get(key).add(id);
        } else {
            index.put(key, new ArrayList<String>() {{
                add(id);
            }});
        }
    }

    public void delete(String id, String key) {
        index.get(key).remove(id);
        if (index.get(key).isEmpty()) {
            index.remove(key);
        }
    }

    public void mod(String id, String key, String oldKey) {
        delete(id, oldKey);
        add(id, key);
    }

    public ArrayList<String> find(String key) {
        return index.get(key);
    }

}
